
import dayjs from 'dayjs'
import weekOfYear from 'dayjs/plugin/weekOfYear'
import quarterOfYear from 'dayjs/plugin/quarterOfYear'

dayjs.extend(weekOfYear)
dayjs.extend(quarterOfYear)

export default function useTimeRange() {
  const getTimeRange = (type:any) => {
    let start, end
    switch(type) {
      case '今天':
        start = dayjs().startOf('day').format('YYYY-MM-DD HH:mm:ss')
        end = dayjs().endOf('day').format('YYYY-MM-DD HH:mm:ss')
        break
      case '本周':
        start = dayjs().startOf('week').format('YYYY-MM-DD HH:mm:ss')
        end = dayjs().endOf('week').format('YYYY-MM-DD HH:mm:ss')
        break
      case '本月':
        start = dayjs().startOf('month').format('YYYY-MM-DD HH:mm:ss')
        end = dayjs().endOf('month').format('YYYY-MM-DD HH:mm:ss')
        break
      case '本年':
        start = dayjs().startOf('year').format('YYYY-MM-DD HH:mm:ss')
        end = dayjs().endOf('year').format('YYYY-MM-DD HH:mm:ss')
        break
      default:
        start = dayjs().startOf('day').format('YYYY-MM-DD HH:mm:ss')
        end = dayjs().endOf('day').format('YYYY-MM-DD HH:mm:ss')
    }

    return { start, end }
  }

  return { getTimeRange }
}
